from functools import wraps

from flask import abort
from flask_login import current_user


def admin_required(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        # 判断当前用户是不是管理员
        if not (current_user.is_active and current_user.is_admin):
            # 如果系统复杂,就需要进行查询数据,判断权限
            abort(403)  # 403 权限错误
        return func(*args, **kwargs)

    return wrapper
